import kmeans


def clusterisationStructure(indice, cube):
	#partitionne le tableau indice en fonction de la structure secondaire des proteines
	tab = [ [] for x in range(8)]
	clusters=[]
	for i in indice:
		nb_helices = float(cube["structure"]["nb_helices"][i])
		nb_coudes = float(cube["structure"]["nb_coudes"][i])
		nb_feuillets = float(cube["structure"]["nb_feuillets"][i])
		
		if(nb_helices==0 and nb_coudes>0 and nb_feuillets>0):
			#sans helices avec coudes et feuillets 
			tab[0].append(i)
		elif(nb_helices==0 and nb_coudes==0 and nb_feuillets>0):
			#sans helices ni coudes avec feuillets
			tab[1].append(i)
		elif(nb_helices==0 and nb_coudes>0 and nb_feuillets==0):
			#sans helices avec coudes sans feuilltes
			tab[2].append(i)
		elif(nb_helices>0 and nb_coudes>0 and nb_feuillets>0):
			#avec helices coudes et feuillets
			tab[3].append(i)
		elif(nb_helices>0 and nb_coudes==0 and nb_feuillets==0):
			#avec helices sans coudes ni feuillets
			tab[4].append(i)
		elif(nb_helices>0 and nb_coudes==0 and nb_feuillets>0):
			#avec helices sans coudes avec feuillets
			tab[5].append(i)
		elif(nb_helices>0 and nb_coudes>0 and nb_feuillets==0):
			#avec helices et coudes sans feuillets
			tab[6].append(i)
		else:
			#sans helices coudes et feuillets
			tab[7].append(i)
	for j in range (len(tab)):
		if len(tab[j])>0:
			clusters.append(tab[j])	
	return clusters